原始题目:剑指 Offer 13. 机器人的运动范围 (opens new window)
解题思路:
可以使用深度优先搜索和剪枝法来做。
- 深度优先搜索:通过递归,往一个方向一直搜索下去,统计格子数,如果格子不符合要求,则结束递归。然后往另一个方向进行搜索。
- 剪枝法:当发现某个各自不符合要求时,就可以提前结束递归。
DFS 函数
递归参数:当前格子位置 、整个方格的边界 、数位和限制 、方格遍历统计数组 。
终止条件: 或者 越界、或者方格已经遍历过、或者 和 的数位和大于 。
递推工作:
- 将 置为 ,表示已经遍历过。
- 可达到格子数 加一。
- 搜索下一个单元格,往右、下方向搜索。
代码:
int ans = 0;
public int movingCount(int m, int n, int k) {
boolean[][] visited = new boolean[m][n];
move(m, n, 0, 0, k, visited);
return ans;
}
private void move(int m, int n, int x, int y, int k, boolean[][] visited){
// 注意参数边界判断
if(x >= m || y >= n || x < 0 || y < 0
|| visited[x][y] || !valid(x, y, k)){
return ;
}
visited[x][y] = true;
ans++;
move(m, n, x+1, y, k, visited);
move(m, n, x, y+1, k, visited);
}
/**
* 判断 x 和 y 的数位和符合要求
* @return 符合返回 true,否则返回 false
*/
private boolean valid(int x, int y, int k){
return k >= (sumOfDigital(x) + sumOfDigital(y));
}
/**
* 计算数位和
*/
private int sumOfDigital(int n){
int sum = 0;
while(n != 0){
sum += n % 10;
n /= 10;
}
return sum;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
复杂度分析
- 时间复杂度 :最多只需要遍历 个方格。
- 空间复杂度 :需要辅助数组 ,空间为 。